#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
收集所有txt文件内容到CSV文件
"""

import os
import csv

def collect_to_csv(folder_path, output_csv):
    """
    将指定文件夹中的所有txt文件内容收集到一个CSV文件中
    
    参数:
        folder_path: txt文件所在的文件夹路径
        output_csv: 输出的CSV文件路径
    """
    print(f"开始收集 {folder_path} 文件夹中的txt文件内容...")
    
    with open(output_csv, 'w', encoding='utf-8', newline='') as f:
        writer = csv.writer(f)
        # 写入CSV头部
        writer.writerow(['文件名', '原始内容', '话术1', '话术2', '话术3', '话术4', '话术5'])
        
        # 获取所有txt文件并排序
        files = [f for f in os.listdir(folder_path) if f.endswith('.txt')]
        files.sort()
        
        file_count = 0
        
        for filename in files:
            file_path = os.path.join(folder_path, filename)
            try:
                with open(file_path, 'r', encoding='utf-8') as txt_file:
                    content = txt_file.read().strip()
                    if content:
                        # 第一列是文件名，第二列是原始内容，第三列也是原始内容（话术1）
                        # 后续列留空，用于手动填写变体话术
                        writer.writerow([filename, content, content, '', '', '', ''])
                        file_count += 1
                        print(f"已处理: {filename}")
                    else:
                        print(f"警告: {filename} 内容为空，已跳过")
            except Exception as e:
                print(f"错误: 处理 {filename} 时出错: {str(e)}")
        
        print(f"收集完成! 总共处理了 {file_count} 个文件，数据已保存到 {output_csv}")

if __name__ == "__main__":
    # 默认参数：01文件夹，输出到sales_scripts.csv
    folder_path = input("请输入txt文件所在的文件夹路径（默认为'01'）: ") or '01'
    output_csv = input("请输入输出CSV文件名（默认为'sales_scripts.csv'）: ") or 'sales_scripts.csv'
    
    collect_to_csv(folder_path, output_csv) 